package com.sudo.boot.common.util.task;

import lombok.Getter;

import java.util.HashMap;
import java.util.Map;

public class BaseTaskRegistry {
    @Getter
    private static final Map<String, Class<?>> beanClassMap = new HashMap<>();

    private BaseTaskRegistry() {
    }

    public static void setModuleClassMap(String key, Class<?> aClass){
        beanClassMap.put(key, aClass);
    }

    public static Class<?> getClassByBean(String beanName){
        return beanClassMap.get(beanName);
    }

}
